/**
 * 代号:隐无为 2017：厚溥
 * 文件名：MyHandler.java
 * 创建人：柯栋
 * 日期：2017年12月19日
 * 修改人：
 * 描述：
 */
package org.kd.websocket;

import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import org.springframework.web.socket.CloseStatus;
import org.springframework.web.socket.TextMessage;
import org.springframework.web.socket.WebSocketHandler;
import org.springframework.web.socket.WebSocketMessage;
import org.springframework.web.socket.WebSocketSession;

/**
 * 用途：消息处理类
 */
public class MyHandler2 implements WebSocketHandler{

private static final List<WebSocketSession> users = new ArrayList<>();
public static final Map<String, WebSocketSession> userSocketSessionMap = new HashMap<String, WebSocketSession>();  
	//用户进入系统监听 当新连接建立的时候，被调用

	@Override
	public void afterConnectionEstablished(WebSocketSession session) throws Exception {
		System.out.println("成功进入了系统。。。");
		users.add(session);
		String uid = (String) session.getAttributes().get("uid");  
        if (userSocketSessionMap.get(uid) == null) {  
            userSocketSessionMap.put(uid, session);  
        }  
		
	}

	// 处理消息  接受前端发送过来的消息
	@Override
	public void handleMessage(WebSocketSession session, WebSocketMessage<?> message) throws Exception {
	    //将消息进行转化，因为是消息是json数据，可能里面包含了发送给某个人的信息，所以需要用json相关的工具类处理之后再封装成TextMessage，
	   // 消息的封装格式一般有{from:xxxx,to:xxxxx,msg:xxxxx}，来自哪里，发送给谁，什么消息等等	    
		  TextMessage msg = new TextMessage(message.getPayload().toString());  
	   System.out.println(msg);
	    //给所有用户群发消息
	    sendMessagesToUsers(msg);
	    // 给当前WebSocketSession发
	    session.sendMessage(new TextMessage("kkkk"));
	    //给指定用户
	    sendMessageToUser2("666",new TextMessage("kkkk666"));
	    
	    

	}
        
        //后台错误信息处理方法
	@Override
	public void handleTransportError(WebSocketSession session, Throwable exception) throws Exception {

	}

	//用户退出后的处理，不如退出之后，要将用户信息从websocket的session中remove掉，这样用户就处于离线状态了，也不会占用系统资源
	@Override
	public void afterConnectionClosed(WebSocketSession session, CloseStatus closeStatus) throws Exception {
		if(session.isOpen()){
			session.close();
		}
		users.remove(session);
		System.out.println("退出系统");
		
	}

	@Override
	public boolean supportsPartialMessages() {
		return false;
	}
	
	/**
	 * 给所有的用户发送消息
	 */
	public void sendMessagesToUsers(TextMessage message){
		for(WebSocketSession user : users){
			try {
			    //isOpen()在线就发送
				if(user.isOpen()){
					user.sendMessage(message);
				}
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
	}
	
	 /** 
     * 给某个用户发送消息 
     * @param userid 
     * @param message 
     * @throws IOException 
     */  
    public void sendMessageToUser2(String userid, TextMessage message)throws IOException {  
        WebSocketSession session = userSocketSessionMap.get(userid);  
        System.out.println(message);  
        if (session != null && session.isOpen()) {  
            session.sendMessage(message);  
        } else {  
            System.out.println("用户："+userid +"websocketSession已失效！");  
        }  
    }  
      
	
	/**
	 * 发送消息给指定的用户
	 */
	public void sendMessageToUser(String userId,TextMessage message){
		for(WebSocketSession user : users){
			if(user.getAttributes().get("666").equals(userId)){
				try {
				    //isOpen()在线就发送
					if(user.isOpen()){
						user.sendMessage(message);
					}
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
		}
	}
	
}
